Fixed the failing test