Fixes small issue