Branch fix.

master had the previous version of the code.