Fix for previuos commit