Fix for beta 4 compatibility
Merge pull request #1 from tonyd256/td-beta4

Fix for beta 4 compatibility