Fixes bug!