Fix minor bug in custom prompt