fix shamt bug