bugfxing
adding test