Fixing test
I hate Java's boxing