Fix code overflow