Fixed the failing test, added comment