I don't know why the last commit worked for me when I tested it.. here's a hack to get the same functionality