1.7.8 update
Updated to work with 1.7.8