Fix command registration