Fixes schema bug