diff --git a/karabo/simulation/sky_model.py b/karabo/simulation/sky_model.py index 23af220a..29d3d7a3 100644 --- a/karabo/simulation/sky_model.py +++ b/karabo/simulation/sky_model.py @@ -870,8 +870,8 @@ def get_OSKAR_sky( :return: oskar sky model """ - if sky.shape[1] > SkyModel.SOURCES_COLS: - return oskar.Sky.from_array(sky[:, : SkyModel.SOURCES_COLS], precision) + if sky.shape[1] > 12: + return oskar.Sky.from_array(sky[:, :12], precision) else: return oskar.Sky.from_array(sky, precision)