Fixes minor bug