Failing test fixed