[STORM-537] Cleaned imports after tests removal