Fix cap bug