Fixes a little bug, adds migration