Fix Bryan's code