Fix use of removed methods.

Signed-off-by: Max Fierke <max@maxfierke.com>