Mały fix, bo pierwszy test nie przechodził. Trochę ryzykowny fix, ale nie chce misie myśleć nad lepszym ;)