Fixed first (and the only) test :)